import sqlite3

# 定义返回类型为 tuple
def select(sql) -> list:
    '''查询数据库'''
    cursor, conn = connect_db()
    cursor.execute(sql)
    res = cursor.fetchall()
    clse_connect(cursor, conn)
    return res

def update(sql):
    '''修改数据库'''
    print(sql)
    cursor, conn = connect_db()
    cursor.execute(sql)
    conn.commit()
    clse_connect(cursor, conn)


def connect_db():
    # 创建或连接到一个数据库文件
    conn = sqlite3.connect("iron.db")  # 如果文件不存在，会自动创建
    # 创建一个游标对象
    cursor = conn.cursor()
    return cursor, conn

def clse_connect(cursor, conn):
    # 关闭连接
    cursor.close()
    conn.close()